python main.py --dimension=512 --angle=180 --backbone='vgg16' --num_worker=8 --train_from_begin=True --start_epoch=0 --batch_size=128 --loss_weight=10.0 --number_of_epoch=100 --learning_rate_val=0.001 --tiny_split=False --mode='train' --cuda=True --project_data_dir='/home/a409/users/lihaowei/my_model/cosplace/' --dataset_dir='/home/a409/users/lihaowei/data/CVUSA_DSM/'
# python main.py --dimension=512 --angle=180 --backbone='vig' --num_worker=8 --train_from_begin=False --start_epoch=0 --batch_size=64 --loss_weight=10.0 --number_of_epoch=150 --learning_rate_val=1e-5 --tiny_split=False --mode='train' --cuda=True --project_data_dir='/home/a409/users/lihaowei/my_model/cosplace/' --dataset_dir='/home/a409/users/lihaowei/data/CVUSA_DSM/'
